Skip to content

During DLRM examples to create prediction using TorchRec library. This shows basic example of how to use TorchRec library quickly locally. #3043

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 1 commit into from

Conversation

yupadhyay
Copy link
Contributor

Differential Revision: D75989524

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Jun 5, 2025
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 9, 2025
… basic example of how to use TorchRec library quickly locally. (pytorch#3043)

Summary:
Pull Request resolved: pytorch#3043

Rollback Plan:

Differential Revision: D75989524
yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 9, 2025
… basic example of how to use TorchRec library quickly locally. (pytorch#3043)

Summary:

Adding end to end example of using TorchRec library with DLRM to create recommendation. You can use this example in local machine to run and test how TorchRec is used for simple use case.

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 9, 2025
… basic example of how to use TorchRec library quickly locally. (pytorch#3043)

Summary:

Adding end to end example of using TorchRec library with DLRM to create recommendation. You can use this example in local machine to run and test how TorchRec is used for simple use case.

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

@yupadhyay yupadhyay self-assigned this Jun 9, 2025
yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 9, 2025
… basic example of how to use TorchRec library quickly locally. (pytorch#3043)

Summary:

Adding end to end example of using TorchRec library with DLRM to create recommendation. You can use this example in local machine to run and test how TorchRec is used for simple use case.

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 9, 2025
… basic example of how to use TorchRec library quickly locally. (pytorch#3043)

Summary:

Adding end to end example of using TorchRec library with DLRM to create recommendation. You can use this example in local machine to run and test how TorchRec is used for simple use case.

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 9, 2025
… basic example of how to use TorchRec library quickly locally. (pytorch#3043)

Summary:
Pull Request resolved: pytorch#3043

Adding end to end example of using TorchRec library with DLRM to create recommendation. You can use this example in local machine to run and test how TorchRec is used for simple use case.

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 9, 2025
… basic example of how to use TorchRec library quickly locally. (pytorch#3043)

Summary:

Adding end to end example of using TorchRec library with DLRM to create recommendation. You can use this example in local machine to run and test how TorchRec is used for simple use case.

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 9, 2025
… basic example of how to use TorchRec library quickly locally. (pytorch#3043)

Summary:
Pull Request resolved: pytorch#3043

Adding end to end example of using TorchRec library with DLRM to create recommendation. You can use this example in local machine to run and test how TorchRec is used for simple use case.

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 10, 2025
… basic example of how to use TorchRec library quickly locally. (pytorch#3043)

Summary:

Adding end to end example of using TorchRec library with DLRM to create recommendation. You can use this example in local machine to run and test how TorchRec is used for simple use case.

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

1 similar comment
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 10, 2025
… basic example of how to use TorchRec library quickly locally. (pytorch#3043)

Summary:
Pull Request resolved: pytorch#3043

Adding end to end example of using TorchRec library with DLRM to create recommendation. You can use this example in local machine to run and test how TorchRec is used for simple use case.

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

1 similar comment
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 11, 2025
… basic example of how to use TorchRec library quickly locally. (pytorch#3043)

Summary:
Pull Request resolved: pytorch#3043

Adding end to end example of using TorchRec library with DLRM to create recommendation. You can use this example in local machine to run and test how TorchRec is used for simple use case.

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

1 similar comment
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 11, 2025
… basic example of how to use TorchRec library quickly locally. (pytorch#3043)

Summary:
Pull Request resolved: pytorch#3043

Adding end to end example of using TorchRec library with DLRM to create recommendation. You can use this example in local machine to run and test how TorchRec is used for simple use case.

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

1 similar comment
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

yupadhyay added a commit to yupadhyay/torchrec that referenced this pull request Jun 11, 2025
…ctions using TorchRec capabilities. (pytorch#3043)

Summary:
Pull Request resolved: pytorch#3043

examples demonstrating how to use TorchRec for recommendation model predictions:

*   **DLRM (Deep Learning Recommendation Model)**: Implementation using TorchRec's EmbeddingBagCollection and KeyedJaggedTensor
*   **Two Tower Model**: Alternative architecture for recommendation systems

Both examples include complete implementations with training, evaluation, and prediction capabilities.

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

…ctions using TorchRec capabilities. (pytorch#3043)

Summary:
Pull Request resolved: pytorch#3043

examples demonstrating how to use TorchRec for recommendation model predictions:

*   **DLRM (Deep Learning Recommendation Model)**: Implementation using TorchRec's EmbeddingBagCollection and KeyedJaggedTensor
*   **Two Tower Model**: Alternative architecture for recommendation systems

Both examples include complete implementations with training, evaluation, and prediction capabilities.

Reviewed By: TroyGarden, aliafzal

Differential Revision: D75989524
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D75989524

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants